"""
Case Type   : mot支持的索引
Case Name   : MOT表支持在总列宽=256的表上创建索引，数据类型为varchar
Create At   : 2020/09/16
Owner       : opentestcase001
Description :
    1.设置enable_incremental_checkpoint参数为off,并重启数据库
    2.创建MOT表，总列宽=256的表上创建索引，数据类型为varchar，清理数据
      键大小包括以字节为单位的列大小+列附加大小，其是维护索引所需的开销(详见表格不同类型对应的列附加大小)
      如果索引不是唯一的，需要额外的8字节
    3.恢复环境默认配置，修改enable_incremental_checkpoint为on，并重启数据库
Expect      :
    1.设置成功，重启数据库成功
    2.创建MOT表成功，索引创建成功，清理环境成功
    3.修改参数成功，重启数据库成功
History     :
    modified by opentestcase042,2023.01.05，优化MOT用例
"""

import os
import unittest
from testcase.utils.Logger import Logger
from testcase.utils.Common import Common
from testcase.utils.CommonSH import CommonSH
from testcase.utils.Constant import Constant


class Mot_DDL_test(unittest.TestCase):

    def setUp(self):
        self.logger = Logger()
        self.primary_sh = CommonSH('PrimaryDbUser')
        self.constant = Constant()
        self.commonsh = CommonSH
        self.common = Common()
        self.config = 'enable_incremental_checkpoint'
        self.logger.info(f'-----{os.path.basename(__file__)} start-----')
        self.logger.info('------------修改配置，并重启数据库------------')
        show_msg = self.common.show_param(self.config)
        if 'off' not in show_msg:
            msg1 = self.primary_sh.execute_gsguc('set',
                                            self.constant.GSGUC_SUCCESS_MSG,
                                            f'{self.config}=off')
            self.logger.info(msg1)
            msg2 = self.primary_sh.restart_db_cluster()
            self.logger.info(msg2)
            msg3 = self.primary_sh.get_db_cluster_status()
            self.assertTrue("Normal" or "Degrade" in msg3)

    def test_mot_none_index(self):
        text = '---------step:创建MOT表，插入数据，清理数据;expect:成功----------'
        self.logger.info(text)
        self.schema = 'schema_mot_0063'
        self.t_name = 't_mot_0063'
        self.t_name_1 = 't_mot_0063_1'
        self.sql_cmd = f'''
                        drop schema if exists {self.schema} cascade;
                        create schema {self.schema};
                        create foreign table {self.schema}.{self.t_name}
                                      (t1 varchar(140) not null,
                                      t2 varchar(100) not null,
                                      t3 varchar(1) not null);
                        create index index1 on {self.schema}.{self.t_name}(t1,t2);
                        create foreign table {self.schema}.{self.t_name_1}
                                      (t1 varchar(247) not null,
                                      t2 varchar(1) not null,
                                      t3 varchar(1) not null);
                        create unique index index2 on {self.schema}.{self.t_name_1}(t1,t3);
                        drop schema {self.schema} cascade;
                     '''
        msg4 = self.primary_sh.execut_db_sql(self.sql_cmd)
        self.logger.info(msg4)
        self.assertIn(self.constant.CREATE_FOREIGN_SUCCESS_MSG, msg4)
        self.assertIn(self.constant.CREATE_INDEX_SUCCESS, msg4)
        self.assertIn(self.constant.DROP_SCHEMA_SUCCESS_MSG, msg4)

    def tearDown(self):
        self.logger.info('------------this is tearDown-------------')
        self.logger.info('---------恢复配置，enable_incremental_checkpoint=on---------')
        show_msg = self.common.show_param(self.config)
        if 'on' not in show_msg:
            msg5 = self.primary_sh.execute_gsguc('set',
                                                 self.constant.GSGUC_SUCCESS_MSG,
                                                 f'{self.config}=on')
            self.logger.info(msg5)
            msg6 = self.primary_sh.restart_db_cluster()
            self.logger.info(msg6)
            msg7 = self.primary_sh.get_db_cluster_status()
            self.assertTrue("Normal" or "Degrade" in msg7)
        show_msg = self.common.show_param(self.config)
        self.assertTrue('on' in show_msg)
        self.logger.info(f'-----{os.path.basename(__file__)} end-----')
